package com.thermofisher.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.FileSystemResource;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.stereotype.Component;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import java.io.File;

@Component
@Slf4j
public class MailUtil
{

    @Autowired
    private JavaMailSender mailSender;

    @Value("${spring.mail.username}")
    private String from;

    public void sendSimpleMail(String to, String subject, String content)
    {
        SimpleMailMessage message = new SimpleMailMessage();
        message.setFrom(from);
        message.setTo(to);
        message.setSubject(subject);
        message.setText(content);
        try
        {
            mailSender.send(message);
            log.info("简单邮件已经发送。");
        }
        catch (Exception e)
        {
            log.error("发送简单邮件时发生异常！", e);
        }
    }

    public void sendHtmlMail(String to, String subject, String content)
    {

        try
        {
            MimeMessage mimeMessage = mailSender.createMimeMessage();
            MimeMessageHelper messageHelper = new MimeMessageHelper(mimeMessage, true, "UTF-8");
            messageHelper.setFrom(from);
            messageHelper.setSubject(subject);
            messageHelper.setTo(to);
            messageHelper.setText(
                    content, true);

            mailSender.send(mimeMessage);
            log.info("邮件发送成功！");
        }
        catch (Exception e)
        {
            log.info("异常信息：" + e);
        }
    }

    public void sendAttachmentsMail(String to, String subject, String content, String filePath)
    {
        try
        {
            MimeMessageHelper helper = getMimeMessageHelper(to, subject, content);
            FileSystemResource file = new FileSystemResource(new File(filePath));
            String fileName = filePath.substring(filePath.lastIndexOf(File.separator));
            helper.addAttachment(fileName, file);
            MimeMessage message = helper.getMimeMessage();
            mailSender.send(message);
            log.info("带附件的邮件已经发送。");
        }
        catch (MessagingException e)
        {
            log.error("发送带附件的邮件时发生异常！", e);
        }
    }

    public void sendInlineResourceMail(String to, String subject, String content, String rscPath,
                                       String rscId)
    {
        try
        {
            MimeMessageHelper helper = getMimeMessageHelper(to, subject, content);
            FileSystemResource res = new FileSystemResource(new File(rscPath));
            helper.addInline(rscId, res);
            MimeMessage message = helper.getMimeMessage();
            mailSender.send(message);
            log.info("嵌入静态资源的邮件已经发送。");
        }
        catch (MessagingException e)
        {
            log.error("发送嵌入静态资源的邮件时发生异常！", e);
        }
    }

    public MimeMessageHelper getMimeMessageHelper(String to, String subject, String content)
    {
        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper helper = null;
        try
        {
            helper = new MimeMessageHelper(message, true);
            helper.setFrom(from);
            helper.setTo(to);
            helper.setSubject(subject);
            helper.setText(content, true);
        }
        catch (MessagingException e)
        {
            e.printStackTrace();
        }
        return helper;
    }
}
